Skip to content

fix(mlops): keep S3CompressionType/S3DataDistributionType in ClarifyCheckStep inputs - #6421

Closed
AkshayShah03 wants to merge 1 commit into
aws:masterfrom
AkshayShah03:fix/clarify-check-step-s3-input-fields
Closed

AkshayShah03 wants to merge 1 commit into
aws:masterfrom
AkshayShah03:fix/clarify-check-step-s3-input-fields

Conversation

@AkshayShah03

Copy link
Copy Markdown

Issue #, if available: N/A

Description of changes:

ClarifyCheckStep builds its dataset ProcessingInput with s3_compression_type and s3_data_distribution_type taken from the DataConfig. However, ClarifyCheckStep.arguments serializes only S3Uri, LocalPath, S3DataType and S3InputMode.

So DataConfig(..., s3_compression_type="Gzip") in a ClarifyCheckStep produces a processing job without S3CompressionType, and the gzipped dataset reaches the Clarify container still compressed. Core's SageMakerClarifyProcessor passes both fields through.

This change serializes S3DataDistributionType and S3CompressionType when they're set, and skips None or unassigned values.

Testing:

  • New test_dataset_compression_and_distribution_type_propagated. It fails before this change.
  • sagemaker-mlops/tests/unit/workflow: 360 passed.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

…heckStep inputs

ClarifyCheckStep builds the dataset ProcessingInput with
s3_compression_type and s3_data_distribution_type from the DataConfig, but
ClarifyCheckStep.arguments only serialized S3Uri, LocalPath, S3DataType and
S3InputMode. A DataConfig(s3_compression_type='Gzip') therefore produced a
processing job without S3CompressionType, so the gzipped dataset reached the
Clarify container still compressed.

Serialize both fields when they are set, as core's SageMakerClarifyProcessor
does.
@AkshayShah03

Copy link
Copy Markdown
Author

Closing in favour of #6406. After #6497 moved the check steps' request building into the shared _build_check_job_request, this fix and the NetworkConfig fix in #6406 change the same helper, so I've folded the S3CompressionType/S3DataDistributionType change and its test into #6406.

This branch had an error being deployed

1 failed deployment
manual-approval — 95318553 Deployed Oct 10, 2026 by AkshayShah03 via wait-for-approval #1973
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant